from config import to_abs_path


class MzConfig:
    def __init__(self):
        self.pretrained_model_path = to_abs_path('model/mengzi/mengzi-bert-base')
        self.model_path = to_abs_path('model/mengzi/mengzi_model.bin')
        self.f1_img_path = to_abs_path('img/mz_f1.png')
        self.log_batch = 5
        self.epochs = 10
        self.lr = 1e-1
        self.batch_size = 400
        self.max_length = 512


m_conf = MzConfig()
